Entropy Regularization: A Free Correction to Cross-Entropy for Verified Demonstrations
熵正则化:对可验证演示交叉熵的免费修正
🧭 LLM 后训练 · 熵正则化
Department of Statistics, University of Michigan
提出 ER-CE,用 token 级熵正则化防止概率质量扩散到未支持的输出,提升可验证任务 verifier 准确率
📌 工作参考价值(4 点)
- 在生成式推荐/文案生成等多正确解可验证场景(如商品标题是否符合规则、推荐理由是否准确),后训练若只用 CE 模仿单一 gold 答案,可能学到次优分布。可尝试在 loss 中加入 token 级熵正则项,实现只需对 logits 计算熵再加权合并,几乎零额外成本,避免模型过度自信于某一路径。
- 若任务有可验证的 reward(如搜索 query 改写后召回率、广告文案离线规则校验),而训练数据每个样本只给一个专家解,直接 CE 可能偏离最大化 verifier 准确率。可借鉴 ER-CE 思路,使策略保留更多样化的正确输出概率,而非只模仿给定演示。
- 注意熵正则化会鼓励输出分布更均匀,可能降低生成确定性,影响线上体验(如推荐文案需风格稳定)。实际落地时需小范围调熵系数,先在线下验证集评估 verifier acc 与多样性的 trade-off。
- 该文主要针对数学/代码,部分思路可迁移,但电商/推荐场景的 verifier 往往较弱或缺失,需先定义好规则校验器再使用此正则化。